#!/usr/bin/env bash

export PYTHONPATH="/raid/xingwg/workspace/boya-reid"

cd ../src
model=resnet101_ibn_a
lr=0.006
epoch=50
python3 ./tools/export_graph.py --config_file=../exp/configs/naic_resnet101_ibn_a.yml \
SOLVER.BASE_LR "(${lr})" \
TEST.WEIGHT "('../exp/naic_a/${model}_384x192_b128_lr${lr}_epoch50_0.9triplet_0.1circle_std/\
${model}_384x192_b128_lr${lr}_epoch50_0.9triplet_0.1circle_std_${epoch}.pth')" \
TEST.RE_RANKING "(True)" \
TEST.NECK_FEAT "('after')" \
TEST.FEAT_NORM "('yes')" \
TEST.FLIP_FEATS "('on')"

python3 -m onnxsim "${model}.onnx" "${model}_sim.onnx"